package com.study.spmongodb.book.wwj.char07;

import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.nio.file.attribute.PosixFilePermission;
import java.nio.file.attribute.PosixFilePermissions;
import java.util.Set;
import java.util.concurrent.TimeUnit;

/**
 * @author MI
 * @version 1.0
 * @date 2020/12/2 21:27
 */
public class PreventDuplicated {

    private final static String LOCK_PATH = "D:/pci";
    private final static String LOCK_FILE = ".lock";

    private final static String PERMISSIONS = "rw-------";


    public static void main(String[] args) throws IOException {

        //注入hook 线程，在程序退出是删除lock文件。
        Runtime.getRuntime().addShutdownHook(new Thread(() -> {
            System.out.println("the program received kill SINGAL");
            getLockFile().toFile().delete();
        }));

        // 检查文件是否存在.lock文件
        checkRunning();
        //简单模拟当前程序正在运行
        for (; ; ) {
            try {
                TimeUnit.MILLISECONDS.sleep(1);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }

    private static void checkRunning() throws IOException {
        final Path path = getLockFile();
        if (path.toFile().exists()) {
            throw new RuntimeException("the program already running");
        }
        //Windows does not support PosixFilePermission.|windows上不支持
        final Set<PosixFilePermission> perms = PosixFilePermissions.fromString(PERMISSIONS);
        Files.createFile(path, PosixFilePermissions.asFileAttribute(perms));
    }

    private static Path getLockFile() {
        return Paths.get(LOCK_PATH, LOCK_FILE);
    }
    /**
     * - hook进程只有在收到退出信号的时候才会被执行，如果在kill的时候使用了参数-9，那么hook线程不会得到执行，进程会立即退出，因此.lock文件将得不到清理。
     * - hook线程中也可以执行一些资源释放的工作，比如关闭文件句柄，socket链接，数据库connection等。
     * - 尽量不要在hook线程中执行一些耗时非常长的操作，因为其会导致程序迟迟不能退出。
     */
}
